diff --git a/ho_basis_3body.jl b/ho_basis_3body.jl
index 8203c54..9afa77f 100644
--- a/ho_basis_3body.jl
+++ b/ho_basis_3body.jl
@@ -17,9 +17,12 @@ c2 = 2
 
 println("No of threads = ", Threads.nthreads())
 
-Es, n1s, l1s, n2s, l2s = get_2p_basis(E_max)
-mask1 = (n2s .== n2s') .&& (l2s .== l2s')
-mask2 = (n1s .== n1s') .&& (l1s .== l1s')
+@time "Basis" Es, n1s, l1s, n2s, l2s = get_2p_basis(E_max)
+@time "Masks" begin
+    mask1 = (n2s .== n2s') .&& (l2s .== l2s')
+    mask2 = (n1s .== n1s') .&& (l1s .== l1s')
+end
+
 println("Basis size = ", length(Es))
 
 println("Constructing KE matrices")